package 剑指offer;

import java.util.HashMap;
import java.util.Map;

/**
 * @author: tyy 剑指 Offer
 * 520. 检测大写字母
我们定义，在以下情况时，单词的大写用法是正确的：
全部字母都是大写，比如 "USA" 。
单词中所有字母都不是大写，比如 "leetcode" 。
如果单词不只含有一个字母，只有首字母大写， 比如 "Google" 。
给你一个字符串 word 。如果大写用法正确，返回 true ；否则，返回 false 。

示例 1：
输入：word = "USA"
输出：true
 * @create: 2021-11-12 17:40
 * @description:
 **/

public class Solution26 {
    private Map<String, Integer> map;
    public Solution26() {
        map = new HashMap<>();
    }

    public void insert(String key, int val) {
        map.put(key, val);
    }

    public int sum(String prefix) {
        int sum = 0;
        for (String s : map.keySet()) {
            if (s.indexOf(prefix) == 0) {
                sum += map.get(s);
            }
        }
        return sum;
    }
    public static void main(String[] args) {
        Solution26 solution26 = new Solution26();
        solution26.insert("ab", 3);
        solution26.insert("abc", 6);
        solution26.insert("a", 1);
        int count = solution26.sum("a");
        System.out.println("count = " + count);
    }
}

/**
 * Your MapSum object will be instantiated and called as such:
 * MapSum obj = new MapSum();
 * obj.insert(key,val);
 * int param_2 = obj.sum(prefix);
 */